# -*- coding: utf-8 -*-
# @Author   : 白云酱
# Data      :
# Software  :pyChrom
# version   :python 3.10
# @File     :driver_init.py

from libs.base_driver import BaseDriver
from libs.operation.browser_operation import BrowserOperation
from libs.operation.expected_operation import ECOption


class DriverInit:
    def __init__(self):
        """操作浏览器驱动方法"""
        self.driver = BaseDriver()

    def to_browser(self):
        """操作浏览器驱动方法"""
        return BrowserOperation(self.driver)

    def to_ec(self):
        """操作元素的方法"""
        return ECOption
